cd ../..

gpus=$1

python -m trainer.BaselineTrainer test \
    --config configs/experiment/baseline.yaml \
    --config configs/model/CMamba.yaml \
    --config configs/dataset/desed.yaml \
    --config configs/trainer/default.yaml \
    --model.exp_name desed/baseline/ \
    --trainer.devices=${gpus} \
    --ckpt_path /home/shaonian/SED/SED/logs/desed/baseline/cmamba_adamW/version_0/checkpoints/epoch191_metric1.3896.ckpt